from langchain_openai import ChatOpenAI
from dotenv import load_dotenv
from langchain_core.prompts import ChatPromptTemplate
import os
load_dotenv('../.env')

# 配置 DeepSeek 的 API 参数
llm = ChatOpenAI(
    api_key=os.getenv("DEEPSEEK_API_KEY"),  # 替换为你的 API Key
    base_url=os.getenv("DEEPSEEK_BASE_URL"),  # DeepSeek API 地址
    model="deepseek-chat",  # 模型名称
    temperature=0.5,  # 控制随机性
)

# 定义对话提示词
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是一个乐于助人的AI助手"),
    ("user", "{input}")
])

# 创建链式调用
chain = prompt | llm

# 调用模型
response = chain.invoke({"input": "中国有多少省份？多少地级市"})
print(response.content)